package cn.tedu.route;

import com.rabbitmq.client.*;

import java.io.IOException;
import java.util.Scanner;
import java.util.concurrent.TimeoutException;

//路由模式中的消费者
public class RouteConsumer {
    //先准备一个main方法
    public static void main(String[] args) throws IOException, TimeoutException {
        //创建连接工厂对象 ConnectionFactory
        ConnectionFactory connectionFactory = new ConnectionFactory();
        connectionFactory.setHost("192.168.8.100");//设置RabbitMQ的IP地址,这里用的是虚拟机里面的rabbitmq
        connectionFactory.setPort(5672);//设置RabbitMQ的端口号,这里5672是默认的端口号，也可以不写
        connectionFactory.setUsername("tom");//设置用户名
        connectionFactory.setPassword("tom");//设置密码
        connectionFactory.setVirtualHost("/");//设置虚拟主机,如果不指定，默认是 “/”

        //1. 创建tcp连接对象connection（通过ConnectionFactory创建）
        Connection connection = connectionFactory.newConnection();

        //2. 创建channel连接对象（通过Connection创建）
        Channel channel = connection.createChannel();

        //3.通过channel创建交换机
        /**
         * 第一个参数：交换机的名称
         * 第二个参数：交换机的类型 ：direct 直连交换机
         */
        channel.exchangeDeclare("routeMode.logs", "direct");

        //4. 通过channel创建队列，空构造就可以创建一个随机队列名，非持久、独占、自动删除的动作
        String queueName = channel.queueDeclare().getQueue();

        //5. 通过channel将交换机绑定到队列中
        System.out.print("请输入绑定键，多个绑定键之间用空格隔开:");
        String str = new Scanner(System.in).nextLine();
        String[] routingKeys = str.split("\\s+");
        for (String routingKey:routingKeys){
            channel.queueBind(queueName,"routeMode.logs",routingKey);
        }

        //6. 消费数据
        //6.1 编写消息处理的回调函数
        DeliverCallback deliverCallback = new DeliverCallback() {
            @Override
            public void handle(String consumerTag, Delivery message) throws IOException {
                //message里面封装了要处理的消息
                byte[] body = message.getBody();//获取消息
                //转成对应数据类型的数据进行处理
                String msg = new String(body);
                System.out.println("要处理"+ msg +"消息数据了");
                System.out.println(msg+"消息处理完毕");
            }
        };
        //6.2 编写取消接收数据的回调函数
        CancelCallback cancelCallback = new CancelCallback() {
            @Override
            public void handle(String consumerTag) throws IOException {

            }
        };
        /**
         * 第1个参数：队列的名称
         * 第2个参数：autoACK 是否自动确认，如果设置为true，则自动确认，如果设置为false，则需要手动确认
         * 第3个参数：deliverCallback  用于消息处理的回调函数
         * 第4个参数：cancelCallback  取消接收数据的回调函数
         */
        channel.basicConsume(queueName,true,deliverCallback,cancelCallback);
    }
}

